package com.nlp.visualization.core.discourse;

import com.hankcs.hanlp.classification.classifiers.IClassifier;

import java.util.List;
import java.util.Map;

/**
 * @author LXM
 * @Title: NLPVisualizationTools
 * @Description:
 * @date 2018/5/9下午2:00
 */
public interface IBayesService {


    /**
     * 对指定目录下的文档做贝叶斯分类器的训练
     *
     * @param filePath
     */
    String trainClassifier(String name, String filePath);

    /**
     * 对用户指定的文本做贝叶斯分类器的训练，返回分类器的持久化模型的MD5码
     */
    String trainClassifier(String name, Map<String, String[]> map);

    /**
     * 对指定的分类器，对文本text做预测
     *
     * @param classifier
     * @param text
     * @return
     */
    Map predict(String modelTag, List<String> readyClassiferText);

}
